package com.mf.controller;

import com.mf.util.OrderSNUtil;
import com.mf.util.StringUtil;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Controller;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;
import org.springframework.web.multipart.MultipartHttpServletRequest;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.*;
import java.text.SimpleDateFormat;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

/**
 * @author: Jiang Runqing
 * @date: 2018-09-02 22:51
 */
@CrossOrigin
@Controller
public class FileController {
    private Logger logger = LoggerFactory.getLogger(this.getClass());

    @Value("${file.filePath}")
    private String filePath;


    @PostMapping(value = "/upload")
    @ResponseBody
    public String upload(@RequestParam("file") MultipartFile file) {
        try {
            if (file.isEmpty()) {
                return "文件为空";
            }
            // 获取文件名
            String fileName = file.getOriginalFilename();
            logger.info("上传的文件名为：" + fileName);
            // 获取文件的后缀名
            String preName = fileName.substring(0, fileName.indexOf("."));
            String suffixName = fileName.substring(fileName.lastIndexOf("."));


            String newFileName = new OrderSNUtil().createOrderSN() + StringUtil.randomKey(6) + suffixName;
            logger.info("文件的后缀名为：" + suffixName + "        文件新名称为：" + newFileName);
            String path = filePath + newFileName;
            File dest = new File(path);
            // 检测是否存在目录
            if (!dest.getParentFile().exists()) {
                dest.getParentFile().mkdirs();// 新建文件夹
            }
            file.transferTo(dest);// 文件写入
            return "上传成功";
        } catch (IllegalStateException e) {
            e.printStackTrace();
        } catch (IOException e) {
            e.printStackTrace();
        }
        return "上传失败";

    }


    @PostMapping(value = "/uploadImage")
    @ResponseBody
    public Map<String, Object> upload(@RequestParam("file") MultipartFile files[]) throws Exception{
        List list = new ArrayList();

        Map<String, Object> result = new HashMap<>();
        String serverUrl = filePath;

        if(files!=null&&files.length>0)
        {
            for(int i = 0;i<files.length;i++)
            {
                try{
                    MultipartFile file =files[i];

                    if (file.isEmpty()) { }

                    //上传文件
                    String suffix = file.getOriginalFilename().substring(file.getOriginalFilename().lastIndexOf("."));

                    logger.info("文件的后缀名为：" + suffix);
                    java.util.Date dt = new java.util.Date(System
                            .currentTimeMillis());

                    SimpleDateFormat fmt = new SimpleDateFormat(
                            "yyyyMMddHHmmssSSS");
                    String time = fmt.format(dt);

                    String newImgName = time + suffix;

                    String path = filePath + newImgName;

                    File dest = new File(path);
                    // 检测是否存在目录
                    if (!dest.getParentFile().exists()) {
                        dest.getParentFile().mkdirs();// 新建文件夹
                    }

                    file.transferTo(dest);// 文件写入
                    String imagePath = serverUrl + newImgName;
                    list.add(imagePath);

                } catch (IllegalStateException e) {
                    e.printStackTrace();
                } catch (IOException e) {
                    e.printStackTrace();
                }
            }
            result.put("code", 1);
            result.put("msg", "success");
            result.put("imageList", list);
            return result;
        }

        result.put("code", 0);
        result.put("msg", "fail");
        return result;
    }


    @PostMapping("/batch")
    public String handleFileUpload(HttpServletRequest request) {
        List<MultipartFile> files = ((MultipartHttpServletRequest) request).getFiles("file");
        MultipartFile file = null;
        BufferedOutputStream stream = null;
        for (int i = 0; i < files.size(); ++i) {
            file = files.get(i);
//            String filePath = "/Users/JiangRunqing/Downloads/";
            if (!file.isEmpty()) {
                try {
                    byte[] bytes = file.getBytes();
                    stream = new BufferedOutputStream(new FileOutputStream(
                            new File(filePath + file.getOriginalFilename())));//设置文件路径及名字
                    stream.write(bytes);// 写入
                    stream.close();
                } catch (Exception e) {
                    stream = null;
                    return "第 " + i + " 个文件上传失败 ==> " + e.getMessage();
                }
            } else {
                return "第 " + i + " 个文件上传失败因为文件为空";
            }
        }
        return "上传成功";
    }

    @GetMapping("/download")
    public String downloadFile(HttpServletResponse response) {
//        Appversion appversionList = appversionService.findAppversionNew();
//        String fileName = appversionList.getFileName();// 文件名
        String fileName = "";// 文件名
        String path = " ";
//        String path = appversionList.getPath();

        if (fileName != null) {
            //设置文件路径
            //String filePath = path+fileName;
            //String pathReplace = path.replace("\\","\\\\");
            //  File file = new File(filePath);
            File file = new File(path, fileName);
            if (file.exists()) {
                response.setContentType("application/x-msdownload;charset=GBK");

//                response.setHeader("Content-disposition", "attachment; filename="+filename);
//                filename = new String(filename.getBytes("GBK"),"ISO8859-1"));//解决中文乱码

                response.setContentType("application/octet-stream");
//                response.setContentType("application/force-download");// 设置强制下载不打开
                response.setHeader("Content-Disposition", "attachment;fileName=" + file.getName());// 设置文件名
                byte[] buffer = new byte[1024];
                FileInputStream fis = null;
                BufferedInputStream bis = null;
                try {
                    fis = new FileInputStream(file);
                    bis = new BufferedInputStream(fis);
                    OutputStream os = response.getOutputStream();
                    int i = bis.read(buffer);
                    while (i != -1) {
                        os.write(buffer, 0, i);
                        i = bis.read(buffer);
                    }
                    return "下载成功";
                } catch (Exception e) {
                    e.printStackTrace();
                } finally {
                    if (bis != null) {
                        try {
                            bis.close();
                        } catch (IOException e) {
                            e.printStackTrace();
                        }
                    }
                    if (fis != null) {
                        try {
                            fis.close();
                        } catch (IOException e) {
                            e.printStackTrace();
                        }
                    }
                }
            }
        }
        return "下载失败";
    }
}

